﻿//////////////////////////////////////////////////////////////////////////////
//
// Copyright © 1998-2024 Glodon Company Limited.
//
// Licensed under the MIT License
//
// Permission is hereby granted, free of charge, to any person obtaining a
// copy of this software and associated documentation files (the “Software”),
// to deal in the Software without restriction, including without limitation
// the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
// sell copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in
// all copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED “AS IS”, WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL
// THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.
//
//////////////////////////////////////////////////////////////////////////////
#include "CmdStatusUtils.h"
#include "IDocument.h"
#include "IEditMode.h"
#include "IUiDocument.h"
#include "IUiDocumentViewManager.h"
#include "ISelection.h"
#include "ISectionBox.h"
#include "IGraphicsNodeReference.h"
#include "UiDocumentViewUtils.h"
#include "EnableCompileWarning_The_LAST_IncludeInCpp.h"
using namespace gcmp;

namespace 
{
    gcmp::IDocument* GetCurrentDBIDoc()
    {
        IUiDocumentViewManager* pUiDocViewMgr = IUiDocumentViewManager::Get();
        DBG_WARN_AND_RETURN_NULLPTR_UNLESS(pUiDocViewMgr, L"pUiDocViewMgr为空",L"GDMPLab",L"2024-03-30");
        gcmp::IUiDocument* pUIDoc = pUiDocViewMgr->GetCurrentUiDocument();
        if (pUIDoc)
        {
            return pUIDoc->GetDbDocument();
        }

        return nullptr;
    }
}

bool CmdStatusUtils::IsInDocumentEnvironment()
{
    return GetCurrentDBIDoc() ? true : false;
}

bool CmdStatusUtils::IsInProjectEnvironment()
{
    IDocument* pDoc = GetCurrentDBIDoc();
    return (pDoc && !pDoc->IsFamilyDocument()) ? true : false;
}

bool CmdStatusUtils::IsInEditMode()
{
    IDocument* pDoc = GetCurrentDBIDoc();
    return (pDoc && IEditMode::GetTopActiveEditMode(pDoc)) ? true : false;
}

bool CmdStatusUtils::IsSelectSectionBox()
{
    IDocument *pDoc = GetCurrentDBIDoc();
    DBG_WARN_AND_RETURN_FALSE_UNLESS(pDoc, L"没有当前文档",L"GDMPLab",L"2024-03-30");

    const GraphicsNodeReferenceOwnerPtrSet& selectedGNodes = ISelection::Get()->GetGraphicsNodeReferences();
    FOR_EACH(nodeReference, selectedGNodes)
    {
        IElement *pElement = pDoc->GetElement(nodeReference->GetElementId());
        if (ISectionBox *pSectionBox = dynamic_cast<ISectionBox*>(pElement))
        {
            return true;
        }
    }
    return false;
}

bool CmdStatusUtils::IsInAnimationEnvironment()
{
    // 临时：动画文档后缀
    static const std::wstring s_suffixAnimDocName = L"-animdoc";

    IDocument* pDoc = UiDocumentViewUtils::GetCurrentDocument();
    if (pDoc)
    {
        return StringUtil::EndWith(pDoc->GetName(), s_suffixAnimDocName);
    }
    return false;
}
